import random
import pandas as pd

def generate_security_code(color):
    """生成单个安全码"""
    return color

def generate_security_codes_list(total_length, green_ratio):
    """生成安全码列表"""
    green_codes = int(total_length * green_ratio)
    yellow_codes = total_length - green_codes
    codes_list = []

    # 生成绿码
    codes_list.extend(['green'] * green_codes)

    # 生成黄码
    codes_list.extend(['yellow'] * yellow_codes)

    # 打乱列表
    random.shuffle(codes_list)
    return codes_list

# 设置随机种子保证可重复性
random.seed(42)

# 生成安全码列表
total_codes = 2000
green_ratio = 0.9
security_codes = generate_security_codes_list(total_codes, green_ratio)

# 创建DataFrame
df_codes = pd.DataFrame({"安全码": security_codes})

# 保存到Excel文件的路径
excel_path = "电子作业安全码列表2.xlsx"
df_codes.to_excel(excel_path, index=False)
